<?php
/**
 * PHPExcel
 *
 * Copyright (c) 2006 - 2013 PHPExcel
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 *
 * @category   PHPExcel
 * @package    PHPExcel_Writer_Excel2007
 * @copyright  Copyright (c) 2006 - 2013 PHPExcel (http://www.codeplex.com/PHPExcel)
 * @license    http://www.gnu.org/licenses/old-licenses/lgpl-2.1.txt	LGPL
 * @version    1.7.9, 2013-06-02
 */


/**
 * PHPExcel_Writer_Excel2007_ContentTypes
 *
 * @category   PHPExcel
 * @package    PHPExcel_Writer_Excel2007
 * @copyright  Copyright (c) 2006 - 2013 PHPExcel (http://www.codeplex.com/PHPExcel)
 */
class PHPExcel_Writer_Excel2007_ContentTypes extends PHPExcel_Writer_Excel2007_WriterPart {
	/**
	 * Write content types to XML format
	 *
	 * @param    PHPExcel $pPHPExcel
	 * @param    boolean  $includeCharts Flag indicating if we should include drawing details for charts
	 *
	 * @return    string                        XML Output
	 * @throws    PHPExcel_Writer_Exception
	 */
	public function writeContentTypes(PHPExcel $pPHPExcel = null, $includeCharts = false) {
		// Create XML writer
		$objWriter = null;
		if ($this->getParentWriter()->getUseDiskCaching())
		{
			$objWriter = new PHPExcel_Shared_XMLWriter(PHPExcel_Shared_XMLWriter::STORAGE_DISK, $this->getParentWriter()->getDiskCachingDirectory());
		}
		else
		{
			$objWriter = new PHPExcel_Shared_XMLWriter(PHPExcel_Shared_XMLWriter::STORAGE_MEMORY);
		}

		// XML header
		$objWriter->startDocument('1.0', 'UTF-8', 'yes');

		// Types
		$objWriter->startElement('Types');
		$objWriter->writeAttribute('xmlns', 'http://schemas.openxmlformats.org/package/2006/content-types');

		// Theme
		$this->_writeOverrideContentType($objWriter, '/xl/theme/theme1.xml', 'application/vnd.openxmlformats-officedocument.theme+xml');

		// Styles
		$this->_writeOverrideContentType($objWriter, '/xl/styles.xml', 'application/vnd.openxmlformats-officedocument.spreadsheetml.styles+xml');

		// Rels
		$this->_writeDefaultContentType($objWriter, 'rels', 'application/vnd.openxmlformats-package.relationships+xml');

		// XML
		$this->_writeDefaultContentType($objWriter, 'xml', 'application/xml');

		// VML
		$this->_writeDefaultContentType($objWriter, 'vml', 'application/vnd.openxmlformats-officedocument.vmlDrawing');

		// Workbook
		$this->_writeOverrideContentType($objWriter, '/xl/workbook.xml', 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet.main+xml');

		// DocProps
		$this->_writeOverrideContentType($objWriter, '/docProps/app.xml', 'application/vnd.openxmlformats-officedocument.extended-properties+xml');

		$this->_writeOverrideContentType($objWriter, '/docProps/core.xml', 'application/vnd.openxmlformats-package.core-properties+xml');

		$customPropertyList = $pPHPExcel->getProperties()->getCustomProperties();
		if (!empty($customPropertyList))
		{
			$this->_writeOverrideContentType($objWriter, '/docProps/custom.xml', 'application/vnd.openxmlformats-officedocument.custom-properties+xml');
		}

		// Worksheets
		$sheetCount = $pPHPExcel->getSheetCount();
		for ($i = 0; $i < $sheetCount; ++$i)
		{
			$this->_writeOverrideContentType($objWriter, '/xl/worksheets/sheet' . ($i + 1) . '.xml', 'application/vnd.openxmlformats-officedocument.spreadsheetml.worksheet+xml');
		}

		// Shared strings
		$this->_writeOverrideContentType($objWriter, '/xl/sharedStrings.xml', 'application/vnd.openxmlformats-officedocument.spreadsheetml.sharedStrings+xml');

		// Add worksheet relationship content types
		$chart = 1;
		for ($i = 0; $i < $sheetCount; ++$i)
		{
			$drawings = $pPHPExcel->getSheet($i)->getDrawingCollection();
			$drawingCount = count($drawings);
			$chartCount = ($includeCharts) ? $pPHPExcel->getSheet($i)->getChartCount() : 0;

			//	We need a drawing relationship for the worksheet if we have either drawings or charts
			if (($drawingCount > 0) || ($chartCount > 0))
			{
				$this->_writeOverrideContentType($objWriter, '/xl/drawings/drawing' . ($i + 1) . '.xml', 'application/vnd.openxmlformats-officedocument.drawing+xml');
			}

			//	If we have charts, then we need a chart relationship for every individual chart
			if ($chartCount > 0)
			{
				for ($c = 0; $c < $chartCount; ++$c)
				{
					$this->_writeOverrideContentType($objWriter, '/xl/charts/chart' . $chart++ . '.xml', 'application/vnd.openxmlformats-officedocument.drawingml.chart+xml');
				}
			}
		}

		// Comments
		for ($i = 0; $i < $sheetCount; ++$i)
		{
			if (count($pPHPExcel->getSheet($i)->getComments()) > 0)
			{
				$this->_writeOverrideContentType($objWriter, '/xl/comments' . ($i + 1) . '.xml', 'application/vnd.openxmlformats-officedocument.spreadsheetml.comments+xml');
			}
		}

		// Add media content-types
		$aMediaContentTypes = array();
		$mediaCount = $this->getParentWriter()->getDrawingHashTable()->count();
		for ($i = 0; $i < $mediaCount; ++$i)
		{
			$extension = '';
			$mimeType = '';

			if ($this->getParentWriter()->getDrawingHashTable()->getByIndex($i) instanceof PHPExcel_Worksheet_Drawing)
			{
				$extension = strtolower($this->getParentWriter()->getDrawingHashTable()->getByIndex($i)->getExtension());
				$mimeType = $this->_getImageMimeType($this->getParentWriter()->getDrawingHashTable()->getByIndex($i)->getPath());
			}
			else if ($this->getParentWriter()->getDrawingHashTable()->getByIndex($i) instanceof PHPExcel_Worksheet_MemoryDrawing)
			{
				$extension = strtolower($this->getParentWriter()->getDrawingHashTable()->getByIndex($i)->getMimeType());
				$extension = explode('/', $extension);
				$extension = $extension[1];

				$mimeType = $this->getParentWriter()->getDrawingHashTable()->getByIndex($i)->getMimeType();
			}

			if (!isset($aMediaContentTypes[$extension]))
			{
				$aMediaContentTypes[$extension] = $mimeType;

				$this->_writeDefaultContentType($objWriter, $extension, $mimeType);
			}
		}

		$sheetCount = $pPHPExcel->getSheetCount();
		for ($i = 0; $i < $sheetCount; ++$i)
		{
			if (count($pPHPExcel->getSheet()->getHeaderFooter()->getImages()) > 0)
			{
				foreach ($pPHPExcel->getSheet()->getHeaderFooter()->getImages() as $image)
				{
					if (!isset($aMediaContentTypes[strtolower($image->getExtension())]))
					{
						$aMediaContentTypes[strtolower($image->getExtension())] = $this->_getImageMimeType($image->getPath());

						$this->_writeDefaultContentType($objWriter, strtolower($image->getExtension()), $aMediaContentTypes[strtolower($image->getExtension())]);
					}
				}
			}
		}

		$objWriter->endElement();

		// Return
		return $objWriter->getData();
	}

	/**
	 * Get image mime type
	 *
	 * @param    string $pFile Filename
	 *
	 * @return    string    Mime Type
	 * @throws    PHPExcel_Writer_Exception
	 */
	private function _getImageMimeType($pFile = '') {
		if (PHPExcel_Shared_File::file_exists($pFile))
		{
			$image = getimagesize($pFile);

			return image_type_to_mime_type($image[2]);
		}
		else
		{
			throw new PHPExcel_Writer_Exception("File $pFile does not exist");
		}
	}

	/**
	 * Write Default content type
	 *
	 * @param    PHPExcel_Shared_XMLWriter $objWriter    XML Writer
	 * @param    string                    $pPartname    Part name
	 * @param    string                    $pContentType Content type
	 *
	 * @throws    PHPExcel_Writer_Exception
	 */
	private function _writeDefaultContentType(PHPExcel_Shared_XMLWriter $objWriter = null, $pPartname = '', $pContentType = '') {
		if ($pPartname != '' && $pContentType != '')
		{
			// Write content type
			$objWriter->startElement('Default');
			$objWriter->writeAttribute('Extension', $pPartname);
			$objWriter->writeAttribute('ContentType', $pContentType);
			$objWriter->endElement();
		}
		else
		{
			throw new PHPExcel_Writer_Exception("Invalid parameters passed.");
		}
	}

	/**
	 * Write Override content type
	 *
	 * @param    PHPExcel_Shared_XMLWriter $objWriter    XML Writer
	 * @param    string                    $pPartname    Part name
	 * @param    string                    $pContentType Content type
	 *
	 * @throws    PHPExcel_Writer_Exception
	 */
	private function _writeOverrideContentType(PHPExcel_Shared_XMLWriter $objWriter = null, $pPartname = '', $pContentType = '') {
		if ($pPartname != '' && $pContentType != '')
		{
			// Write content type
			$objWriter->startElement('Override');
			$objWriter->writeAttribute('PartName', $pPartname);
			$objWriter->writeAttribute('ContentType', $pContentType);
			$objWriter->endElement();
		}
		else
		{
			throw new PHPExcel_Writer_Exception("Invalid parameters passed.");
		}
	}
}
